package tianhao.luo.algorithm;

/**
 * KMP算法
 *
 * @author: tianhao.luo@hand-china.com 2021/7/2  下午7:51
 */
public class KMPAlgorithm {

    /**
     * 原串
     */
    private String str1;
    /**
     * 匹配的串
     */
    private String str2;


    public KMPAlgorithm(String str1, String str2) {
        this.str1 = str1;
        this.str2 = str2;
    }

    /**
     * @return 如果是-1就没有匹配到,否贼返回第一个匹配的位置
     */
    public int kmpSearch() {
        int[] next = kmpNext(str2);
        for (int i = 0, j = 0; i < str1.length(); i++) {
            // str1.charAt(i) != str2.charAt(j)时,调整j的位置
            // KMP核心
            while (j > 0 && str1.charAt(i) != str2.charAt(j)){
                j = next[j-1];
            }

            if (str1.charAt(i) == str2.charAt(j)){
                j++;
            }
            if (j == str2.length()){
                return i-j+1;
            }
        }
        return -1;
    }

    /**
     * 获取串的部分匹配值
     */
    public int[] kmpNext(String dest) {
        // 创建一个next数组保存部分匹配值
        int[] next = new int[dest.length()];

        // 如果字符串只有一个字符,它的部分匹配值就是0
        next[0] = 0;
        // j记录扫描时出现和开始重复的字符个数
        for (int i = 1, j = 0; i < dest.length(); i++) {
            // 当dest.charAt(i) != dest.charAt(j)满足时,我们需要从next[j-1]获取新的j,直到我们发现dest.charAt(i) == dest.charAt(j)时退出
            // KMP核心
            while (j > 0 && dest.charAt(i) != dest.charAt(j)) {
                j = next[j - 1];
            }

            // 当dest.charAt(i) == dest.charAt(j)满足时,部分匹配值就是+1
            if (dest.charAt(i) == dest.charAt(j)) {
                j++;
            }
            next[i] = j;
        }
        return next;
    }
}
